# setup.py 示例
from setuptools import setup, find_packages
import torch
import os
import time
from PIL import Image
from tqdm import tqdm
from torch.utils.data import Dataset
from torchvision import transforms
import torch.nn as nn
import torch.nn.functional as F
from torchvision.models import vit_b_16
from einops import rearrange, repeat
from collections import defaultdict
from torch.utils.data import DataLoader  # 新增导入
import numpy as np
from pathlib import Path
import matplotlib.pyplot as plt
from collections import Counter
from sklearn.metrics import average_precision_score  # 新增导入


setup(
    name="vehicle_reid",
    version="0.1",
    packages=find_packages(),
    install_requires=[
        "torch>=2.0",
        "torchvision>=0.15",
        "einops>=0.6"
    ]
)